package com.wupaas.boot.core.common.util;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
 * @author JueYue
 * @date 2021-03-21-3-15
 * @since 1.0
 */
public class CookieUtil {
    // 默认缓存时间,单位/秒, 2H
    private static final int COOKIE_MAX_AGE = 60 * 60 * 2;
    // 保存路径,根路径
    private static final String COOKIE_PATH = "/";

    /**
     * 保存
     * @param response
     * @param key
     * @param value
     * @param ifRemember
     */
    public static void set(HttpServletResponse response, String key, String value, boolean ifRemember) {

        int age = COOKIE_MAX_AGE;
        if (ifRemember) {
            age = COOKIE_MAX_AGE;
        } else {
            age = -1;
        }

        Cookie cookie = new Cookie(key, value);
        cookie.setMaxAge(age);				// Cookie过期时间,单位/秒
        cookie.setPath(COOKIE_PATH);		// Cookie适用的路径
        response.addCookie(cookie);
    }

    /**
     * 保存
     * @param response
     * @param key
     * @param value
     * @param maxAge
     */
    private static void set(HttpServletResponse response,
                            String key, String value, int maxAge, String path) {
        Cookie cookie = new Cookie(key, value);
        cookie.setMaxAge(maxAge);	// Cookie过期时间,单位/秒
        cookie.setPath(path);		// Cookie适用的路径
        response.addCookie(cookie);
    }

    /**
     * 查询value
     * @param request
     * @param key
     * @return
     */
    public static String getValue(HttpServletRequest request, String key) {
        Cookie cookie = get(request, key);
        if (cookie != null) {
            return cookie.getValue();
        }
        return null;
    }

    /**
     * 查询Cookie
     * @param request
     * @param key
     */
    private static Cookie get(HttpServletRequest request, String key) {
        Cookie[] arr_cookie = request.getCookies();
        if (arr_cookie != null && arr_cookie.length > 0) {
            for (Cookie cookie : arr_cookie) {
                if (cookie.getName().equals(key)) {
                    return cookie;
                }
            }
        }
        return null;
    }

    /**
     * 删除Cookie
     * @param request
     * @param response
     * @param key
     */
    public static void remove(HttpServletRequest request, HttpServletResponse response, String key) {
        Cookie cookie = get(request, key);
        if (cookie != null) {
            set(response, key, "", 0, COOKIE_PATH);
        }
    }

}
